package leetcode.editor.cn;

/**
 * ******************************************************
 *
 * @author liugh9
 * @version 1.0
 * @classname UF
 * @description 并查集（UNION-FIND）算法
 *  动态连通性其实可以抽象成给一幅图连线。比如下面这幅图，总共有 10 个节点，他们互不相连，分别用 0~9 标记
 * @date 2023/04/23 17:07
 * <p>
 * ******************************************************
 */
public class UF {

    // 连通分量个数
    private int count;
    // 存储每个节点的父节点
    private int[] parent;

    public UF(int n) {
        this.count = n;
        this.parent = new int[n];
        for (int i = 0; i < n; i++) {
            parent[i] = i;
        }
    }

    /* 将 p 和 q 连接 */
    public void union(int p, int q) {
        int rootP = find(p);
        int rootQ = find(q);
        if (rootP == rootQ)
            return;
        this.parent[rootP] = rootQ;
        // 两个连通分量合并成一个连通分量
        count--;
    }
    /* 判断 p 和 q 是否连通 */
    public boolean connected(int p, int q) {
        int rootP = find(p);
        int rootQ = find(q);
        return rootP == rootQ;
    }

    /* 返回图中有多少个连通分量 */
    public int count() {
        return count;
    }

    /**
     * 找到x节点的根节点
     * @param x
     * @return
     */
    private int find(int x) {
        while (this.parent[x] != x) {
            // 压缩路径
            this.parent[x] = find(this.parent[x]);
            x = this.parent[x];
        }
        return x;
    }
}
